from writeai import writerAI
context = "张三一睁开眼，眼前一暗后缓慢的适应，旁边是一个破旧的操作台，自己躺在床架上。"

data = writerAI(context,1,20)

print(data)